package com.it.product.service.impl;

import com.it.product.service.TestService;
import com.spzx.common.core.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Slf4j
@Service
public class Test002ServiceImpl implements TestService {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 问题1 业务出现异常 锁无法释放
     * 解决措施：给key设置过期时间
     * 新问题2 设置过期时间后 可能会发生误删除 （删除不是自己的锁）
     * 解决措施 删除之前判断是否是自己的锁  给value设置一个随机的
     * 新问题3 删除不是原子性 需要保障判断和删除的原子性
     * 解决措施 使用LUA脚本(是原子性的) 执行判断和删除
     * 新问题4 设置key过期时间过短 程序没有执行完毕 锁先释放了
     * 解决措施 自动续期（看门狗机制） 在2里面
     */
    @Override
    public void testLock() {
        //设置一个随机不重复的作为value防止误删除
        String uuid = UUID.randomUUID().toString().replaceAll("-", "");
        //redis中的分布式锁 同时只有一个对象能抢到锁
        Boolean aBoolean = stringRedisTemplate.opsForValue().setIfAbsent("lock", uuid, 30, TimeUnit.SECONDS);

        if (aBoolean) {
            //抢到锁 执行业务代码
            String value = stringRedisTemplate.opsForValue().get("num");

            // 没有该值return
            if (StringUtils.isBlank(value)) {
                return;
            }
            // 有值就转成成int
            int num = Integer.parseInt(value);
            // 把Redis中的num值+1
            this.stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

            //锁延期
            renewTime("lock", uuid, 30L);

            try {
                TimeUnit.SECONDS.sleep(60);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

            //释放锁
            //stringRedisTemplate.delete("lock");
            //使用LUA脚本保证原子性判断删除
            //创建脚本对象 DefaultRedisScript 泛型脚本语言返回类型 Long 0:失败 1：成功
            DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
            String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1]\n" +
                    "then\n" +
                    "    return redis.call(\"del\",KEYS[1])\n" +
                    "else\n" +
                    "    return 0\n" +
                    "end";
            redisScript.setScriptText(script);
            redisScript.setResultType(Long.class);
            //执行脚本
            stringRedisTemplate.execute(redisScript, Arrays.asList("lock"), uuid);
        } else {
            //没有抢到锁 自旋
            try {
                Thread.sleep(1000);
                this.testLock();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

        }

    }

    /**
     * 锁延期
     * 线程等待超时时间的2/3时间后,执行锁延时代码,直到业务逻辑执行完毕,因此在此过程中,其他线程无法获取到锁,保证了线程安全性
     *
     * @param lockName
     * @param expire   单位：毫秒
     */
    /*private void renewTime(String lockName, String uuid, Long expire) {
        String script = "if(redis.call('hexists', KEYS[1], ARGV[1]) == 1) then redis.call('expire', KEYS[1], ARGV[2]); return 1; else return 0; end";
        new Thread(() -> {
            while (stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class), Arrays.asList(lockName), uuid, expire.toString())) {
                log.info("延期执行：｛｝，｛｝", lockName, uuid);
                try {
                    // 到达过期时间的2/3时间，自动续期
                    Thread.sleep(expire * 1000 / 3);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();
    }*/

    //TODO string类型的lua脚本
    private void renewTime(String lockName, String uuid, Long expire) {
        // Lua 脚本，用于检查字符串是否存在并续期
        String script = "if redis.call('exists', KEYS[1]) == 1 then redis.call('expire', KEYS[1], ARGV[1]); return 1; else return 0; end";

        new Thread(() -> {
            while (stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class),
                    Arrays.asList(lockName), expire.toString())) {
                log.info("延期执行：｛｝，｛｝", lockName, uuid);
                try {
                    // 到达过期时间的2/3时间，自动续期
                    Thread.sleep(expire * 1000 / 3);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();
    }


}